package algorithm_primary.studyMySelf.初级算法.双指针;

/**
 * @author mengHao.heng
 * @date 2023/5/22 16:26
 * @description 三五搜索插入位置  35题
 * 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
 * <p>
 * 请必须使用时间复杂度为 O(log n) 的算法。
 * 输入: nums = [1,3,5,6], target = 5
 * 输出: 2
 */
public class 三五搜索插入位置 {
    public static void main(String[] args) {
        int[] nums = {1, 3};
        System.out.println(searchInsert(nums, 2));
    }

    /**
     * 二分查找  使用双指针指向距离target最近的数值 并且这个数值是小于等于target
     * 如果当前数值等于target那么mid赋值给ans  如果小于也把mid赋值给ans
     * 但是如果下一次不小于了那么这时候肯定不是等于一个数  因为是有序的  所以这时候最近这个ans就是插入的地方
     * @param nums
     * @param target
     * @return
     */
    public static int searchInsert(int[] nums, int target) {
        if (nums.length == 0 || nums[0] >= target) {
            return 0;
        } else if (nums[nums.length - 1] <= target) {
            return nums[nums.length - 1] == target ? nums.length - 1 : nums.length;
        }
        int r = nums.length -1,l = 0,ans = 0;
        while (l <= r) {
            int mid = ((r - l) >> 1) + l;
            if (nums[mid] < target){
                l = mid + 1;
            }else if (nums[mid] >= target){
                ans = mid;
                r = mid - 1;
            }
        }
        return ans;
    }
}
